package com.gzhryc.common.secruity;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Random;

import javax.imageio.ImageIO;

/**
 * 验证码生成工具
 * <h4>使用方法：</h4>
 * 禁止图像缓存 </br>
 * response.setHeader("Pragma", "no-cache");  </br>   
 * response.setHeader("Cache-Control", "no-cache");  </br>
 * response.setDateHeader("Expires", 0);  </br>
 * 将图片输入到响应输出流中 </br>
 * response.setContentType("image/jpeg");</br>
 * ImageIO.write(buffImg, "jpeg", response.getOutputStream()); 
 * @author 张浩
 * @version 0.3.0
 *
 */
public class VerifyCodeTools {

	/** codeSequence 表示字符允许出现的序列值 */
	private static char[] codeSequence = { 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J',
			'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W',
			'X', 'Y', 'Z', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9' };
	
	public static VerifyCode createImg(int width,int height, int codeCount){
		VerifyCode result = new VerifyCode();
		//width-4 除去左右多余的位置，使验证码更加集中显示，减得越多越集中。  
        //codeCount+1     //等比分配显示的宽度，包括左右两边的空格  
        int codeX = (width-4) / (codeCount+1);
        int codeY = height - 7;   
        //height - 10 集中显示验证码  
        int fontHeight = height - 10;    
       
		 // 定义图像buffer     
        BufferedImage buffImg = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);     
        Graphics2D gd = buffImg.createGraphics();     
        // 创建一个随机数生成器类     
        Random random = new Random();     
        // 将图像填充为白色     
        gd.setColor(Color.LIGHT_GRAY);     
        gd.fillRect(0, 0, width, height);     
        // 创建字体，字体的大小应该根据图片的高度来定。     
        Font font = new Font("Fixedsys", Font.PLAIN, fontHeight);     
        // 设置字体。     
        gd.setFont(font);     
        // 画边框。     
        gd.setColor(Color.BLACK);     
        gd.drawRect(0, 0, width - 1, height - 1);     
        // 随机产生160条干扰线，使图象中的认证码不易被其它程序探测到。     
        gd.setColor(Color.gray);     
        for (int i = 0; i < 16; i++) {     
            int x = random.nextInt(width);     
            int y = random.nextInt(height);     
            int xl = random.nextInt(12);     
            int yl = random.nextInt(12);     
            gd.drawLine(x, y, x + xl, y + yl);     
        }     
        // randomCode用于保存随机产生的验证码，以便用户登录后进行验证。     
        StringBuffer randomCode = new StringBuffer();     
        int red = 0, green = 0, blue = 0;
        // 随机产生codeCount数字的验证码。     
        for (int i = 0; i < codeCount; i++) {     
            // 得到随机产生的验证码数字。     
            String strRand = String.valueOf(codeSequence[random.nextInt(36)]);     
            // 产生随机的颜色分量来构造颜色值，这样输出的每位数字的颜色值都将不同。     
            red = random.nextInt(255);     
            green = random.nextInt(255);     
            blue = random.nextInt(255);     
            // 用随机产生的颜色将验证码绘制到图像中。     
            gd.setColor(new Color(red,green,blue));     
            gd.drawString(strRand, (i + 1) * codeX, codeY);      
            randomCode.append(strRand);     
        } 
        //释放资源
        gd.dispose();
        result.setCode(randomCode.toString());
        result.setImg(buffImg);
        
        return result;
	}
	
	public static class VerifyCode {
		
		private String code;
		private BufferedImage img;
		
		public String getCode() {
			return code;
		}
		public void setCode(String code) {
			this.code = code;
		}
		public BufferedImage getImg() {
			return img;
		}
		public void setImg(BufferedImage img) {
			this.img = img;
		}
		
		public byte[] getBytes(){
			try {
				ByteArrayOutputStream out = new ByteArrayOutputStream();
				if(ImageIO.write(img, "jpg", out)){
					return out.toByteArray();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
			return null;
		}
	}
	
}
